package com.example.pojo.mysql;

import com.fasterxml.jackson.annotation.JsonInclude;
import lombok.Data;

import java.io.Serializable;
import java.util.List;
import java.util.stream.Collectors;

/**
 * Treeselect树结构实体类
 *
 * @author
 */
@Data
public class TreeSelect<E> implements Serializable {

    /**
     * 节点ID
     */
    private Long id;

    /**
     * 节点名称
     */
    private String label;

    /**
     * 子节点
     */
    @JsonInclude(JsonInclude.Include.NON_EMPTY)
    private List<TreeSelect> children;

    public TreeSelect() {

    }


    public TreeSelect(E e) {

        if (e instanceof SysDept) {
            SysDept dept = (SysDept) e;
            this.id = dept.getDeptId();
            this.label = dept.getDeptName();
            this.children = dept.getChildren().stream().map(TreeSelect::new).collect(Collectors.toList());
        }

        if (e instanceof SysMenu) {
            SysMenu menu = (SysMenu) e;
            this.id = menu.getMenuId();
            this.label = menu.getMenuName();
            this.children = menu.getChildren().stream().map(TreeSelect::new).collect(Collectors.toList());
        }


    }


}
